package cn.itsource.pet.service.impl;

import cn.itsource.pet.domain.PetType;
import cn.itsource.pet.mapper.PetTypeMapper;
import cn.itsource.pet.service.IPetTypeService;
import cn.itsource.basic.base.service.impl.BaseServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author huang
 * @since 2022-07-18
 */
@Service
public class PetTypeServiceImpl extends BaseServiceImpl<PetType> implements IPetTypeService {

    @Autowired
    private PetTypeMapper petTypeMapper;

    @Override
    public List<PetType> queryTree() {
        //查询出所有数据
        List<PetType> allPetType = petTypeMapper.loadAll();

        // 把所有的数据转换成Map结构--后面查找自己的上级使用
        HashMap<Long, PetType> petTypeMap = new HashMap<>();
        for(PetType petType : allPetType){
            petTypeMap.put(petType.getId(), petType);
        }
        //准备要返回的无限极树，把顶级部门全放到fistPetType集合中
        List<PetType> firstPetType = new ArrayList<>();

        for (PetType petType : allPetType) {
            //1.从所有数据中找出第一层级 firstDept   我们需要返回的数据
            if(petType.getPid() == null || petType.getPid().intValue() == 0 ){
                firstPetType.add(petType);//最顶层的数据
            }else{
                //这里拿到的是有爹的部门
                PetType dadPetType = petTypeMap.get(petType.getPid());
                //判断爹时候存在
                if(dadPetType != null){
                    //如果存在，把自己装到爹的children里
                    dadPetType.getChildren().add(petType);
                }

            }
        }
        return firstPetType;
    }
}
